source activate ke2torch23cu121

date=2025-03-26

# 默认参数
method=ICE
method_type=kl
cnt=1
# Qwen-7B-Chat Qwen1.5-7B-Chat Qwen2-7B-Instruct Qwen2.5-7B-Instruct
# 此版本pre-edit文件使用huggingface版本的KnowEdit数据集

gpu_id=0
for model in Qwen1.5-0.5B-Chat Qwen1.5-1.8B-Chat Qwen1.5-4B-Chat; do
    data_dir=./dataset/KnowEdit/benchmark/ZsRE/ZsRE-test-all.json
    datatype=zsre
    echo ${model}-${datatype}-pre_edit-${cnt}
    CUDA_VISIBLE_DEVICES=${gpu_id} python examples/run_knowedit_llama2.py \
        --editing_method=${method} \
        --hparams_dir=./hparams/${method}/${model}-${method_type}.yaml \
        --data_dir=${data_dir} \
        --datatype=${datatype} \
        --metrics_save_dir=./results/${model}/${method}-${method_type} \
        --pre_file=./pre_edit/${model}_${datatype}_pre_edit.json \
        > log/pre_edit/${date}-${model}-${datatype}-${cnt}.log 2>&1 &
    wait $!  # 等待gpu_id号GPU上的当前任务结束
done &
gpu_id=1
for model in Qwen1.5-0.5B-Chat Qwen1.5-1.8B-Chat Qwen1.5-4B-Chat; do
    data_dir=./dataset/KnowEdit/benchmark/wiki_counterfact/test_cf.json
    datatype=counterfact
    echo ${model}-${datatype}-pre_edit-${cnt}
    CUDA_VISIBLE_DEVICES=${gpu_id} python examples/run_knowedit_llama2.py \
        --editing_method=${method} \
        --hparams_dir=./hparams/${method}/${model}-${method_type}.yaml \
        --data_dir=${data_dir} \
        --datatype=${datatype} \
        --metrics_save_dir=./results/${model}/${method}-${method_type} \
        --pre_file=./pre_edit/${model}_${datatype}_pre_edit.json \
        > log/pre_edit/${date}-${model}-${datatype}-${cnt}.log 2>&1 &
    wait $!  # 等待gpu_id号GPU上的当前任务结束
done &
gpu_id=2
for model in Qwen1.5-0.5B-Chat Qwen1.5-1.8B-Chat Qwen1.5-4B-Chat; do
    data_dir=./dataset/KnowEdit/benchmark/wiki_recent/recent_test.json
    datatype=recent
    echo ${model}-${datatype}-pre_edit-${cnt}
    CUDA_VISIBLE_DEVICES=${gpu_id} python examples/run_knowedit_llama2.py \
        --editing_method=${method} \
        --hparams_dir=./hparams/${method}/${model}-${method_type}.yaml \
        --data_dir=${data_dir} \
        --datatype=${datatype} \
        --metrics_save_dir=./results/${model}/${method}-${method_type} \
        --pre_file=./pre_edit/${model}_${datatype}_pre_edit.json \
        > log/pre_edit/${date}-${model}-${datatype}-${cnt}.log 2>&1 &
    wait $!  # 等待gpu_id号GPU上的当前任务结束
done &
gpu_id=3
for model in Qwen1.5-0.5B-Chat Qwen1.5-1.8B-Chat Qwen1.5-4B-Chat; do
    data_dir=./dataset/KnowEdit/benchmark/WikiBio/wikibio-test-all.json
    datatype=wikibio
    echo ${model}-${datatype}-pre_edit-${cnt}
    CUDA_VISIBLE_DEVICES=${gpu_id} python examples/run_knowedit_llama2.py \
        --editing_method=${method} \
        --hparams_dir=./hparams/${method}/${model}-${method_type}.yaml \
        --data_dir=${data_dir} \
        --datatype=${datatype} \
        --metrics_save_dir=./results/${model}/${method}-${method_type} \
        --pre_file=./pre_edit/${model}_${datatype}_pre_edit.json \
        > log/pre_edit/${date}-${model}-${datatype}-${cnt}.log 2>&1 &
    wait $!  # 等待gpu_id号GPU上的当前任务结束
done &


# 等待所有后台任务完成
wait
